from torchvision import datasets
import torch
import torch.nn as nn
import numpy as np
import matplotlib.pyplot as plt
from torch.optim import SGD, Adam
from torch.utils.data import Dataset,DataLoader
from torchsummary import summary
device = 'cuda'

data_folder = 'E:/study_code/torch_study/aigc/data'
fmnist = datasets.FashionMNIST(data_folder,download=True,train=True)
#
tr_images = fmnist.data
tr_targets = fmnist.targets
val_fmnist = datasets.FashionMNIST(data_folder, download=True, train=False)
val_images = val_fmnist.data
val_targets = val_fmnist.targets

from imgaug import augmenters as iaa

aug = iaa.Affine(scale=2)
plt.imshow(aug.augment_image(tr_images[0]))
plt.title('Scaled image')
plt.show()